vector operations in python